#include <iostream>
#include <signal.h>
#include "../comm/httplib.h"
#include "./oj_controller.hpp"

using namespace httplib;
using namespace ns_controller;

static Controller* ptr = nullptr;

void Recovery(int signo)
{
    ptr->RecoveryMachine();
}

int main()
{
    signal(SIGINT, Recovery);

    // 用户请求的服务路由功能
    Server svr;
    Controller ctrl;

    ptr = &ctrl;

    // 获取所有的题目列表
    svr.Get("/all_questions", [&ctrl](const Request& req, Response& resp){
        // 返回一张包含所有题目的 html 网页
        std::string html;
        ctrl.AllQuestions(&html);
        resp.set_content(html, "text/html; charset=utf-8");
    });

    // 根据题目编号获取题目内容
    // \d 匹配数字, + 匹配任意个字符
    // /qustion/100 -> 正则匹配
    // R"()"": 原始字符串, 保存字符串的原貌, 不进行转义
    svr.Get(R"(/question/(\d+))", [&ctrl](const Request& req, Response& resp){
        std::string number = req.matches[1]; // 题目编号
        std::string html;
        ctrl.OneQuestion(number, &html);
        resp.set_content(html, "text/html; charset=utf-8");
    });

    // 判题功能
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request& req, Response& resp){
        std::string number = req.matches[1]; // 题目编号
        std::string result_json;
        ctrl.Judge(number, req.body, &result_json);
        resp.set_content(result_json, "application/json; charset=utf-8");
    });

    svr.set_base_dir("./wwwroot"); // 设置网站首页
    svr.listen("0.0.0.0", 8080);

    return 0;
}